/*
 *	Copyright (C) 2003-2006 Gabest
 *	http://www.gabest.org
 *
 *  This Program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2, or (at your option)
 *  any later version.
 *
 *  This Program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with GNU Make; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 *  http://www.gnu.org/copyleft/gpl.html
 *
 */

#pragma once

#include "ISubPic.h"

// CDX9SubPic

class CVirtualLock
{
public:
    virtual void Lock() = 0;
    virtual void Unlock() = 0;
};

typedef void (FLock)(void *_pLock);

class CScopeLock
{
    void *m_pLock;
    FLock *m_pUnlockFunc;
public:

    template <typename t_Lock>
    class TCLocker
    {
    public:
        static void fs_Locker(void *_pLock)
        {
            ((t_Lock *)_pLock)->Unlock();
        }
    };

    template <typename t_Lock>
    CScopeLock(t_Lock &_Lock)
    {
        _Lock.Lock();
        m_pLock = &_Lock;
        m_pUnlockFunc = TCLocker<t_Lock>::fs_Locker;
    }

    ~CScopeLock()
    {
        m_pUnlockFunc(m_pLock);
    }
};


class CDX9SubPicAllocator;
class CDX9SubPic : public ISubPicImpl
{
    CComPtr<IDirect3DSurface9> m_pSurface;

protected:
    STDMETHODIMP_(void*) GetObject(); // returns IDirect3DTexture9*

public:
    CDX9SubPicAllocator *m_pAllocator;
    CDX9SubPic(IDirect3DSurface9* pSurface, CDX9SubPicAllocator *pAllocator);
    ~CDX9SubPic();

    // ISubPic
    STDMETHODIMP GetDesc(SubPicDesc& spd);
    STDMETHODIMP CopyTo(ISubPic* pSubPic);
    STDMETHODIMP ClearDirtyRect(DWORD color);
    STDMETHODIMP Lock(SubPicDesc& spd);
    STDMETHODIMP Unlock(RECT* pDirtyRect);
    STDMETHODIMP AlphaBlt(RECT* pSrc, RECT* pDst, SubPicDesc* pTarget);
};

// CDX9SubPicAllocator

class CDX9SubPicAllocator : public ISubPicAllocatorImpl, public CCritSec
{
    CComPtr<IDirect3DDevice9> m_pD3DDev;
    CSize m_maxsize;


    bool Alloc(bool fStatic, ISubPic** ppSubPic);

public:
    static CCritSec ms_SurfaceQueueLock;
    CAtlList<CComPtr<IDirect3DSurface9> > m_FreeSurfaces;
    CAtlList<CDX9SubPic *> m_AllocatedSurfaces;

    void GetStats(int &_nFree, int &_nAlloc);

    CDX9SubPicAllocator(IDirect3DDevice9* pD3DDev, SIZE maxsize, bool fPow2Textures);
    ~CDX9SubPicAllocator();
    void ClearCache();

    // ISubPicAllocator
    STDMETHODIMP ChangeDevice(IUnknown* pDev);
    STDMETHODIMP SetMaxTextureSize(SIZE MaxTextureSize);
};

[uuid("CD6D2AA5-20D3-4ebe-A8A9-34D3B00CC253")]
interface ISubRenderCallback :
public IUnknown
{
    // NULL means release current device, textures and other resources
    STDMETHOD(SetDevice)(IDirect3DDevice9 *dev) = 0;

    // destination video rectangle, will be inside (0, 0)-(width, height)
    // width,height is the size of the entire output window
    STDMETHOD(Render)(REFERENCE_TIME rtStart,
    int left, int top, int bottom, int right,
    int width, int height) = 0;
};

[uuid("E602585E-C05A-4828-AC69-AF92997F2E0C")]
interface ISubRenderCallback2 :
public ISubRenderCallback
{
    STDMETHOD(RenderEx)(REFERENCE_TIME rtStart, REFERENCE_TIME rtStop,
    REFERENCE_TIME AvgTimePerFrame,
    int left, int top, int right, int bottom,
    int width, int height) = 0;
};

[uuid("9CC7F9F7-3ED1-493c-AF65-527EA1D9947F")]
interface ISubRender :
public IUnknown
{
    STDMETHOD(SetCallback)(ISubRenderCallback *cb) = 0;
};
